# !/usr/bin/env python3
# -*- coding: utf-8 -*-
# @Time : 2023/4/25 16:35
# @Author : xyjiang1216
# @Email : jiangxuyan@nudt.edu.cn
# @File : itp_algo_input.py
# @Version: OpenPlanner2.0
from OpenPlanner.core.api.algo_input_class import Link, Stream


class ITPLink(Link):
    def __init__(self, **kwargs):
        super().__init__(**kwargs)
        self.queue_size = kwargs['queue_size']


class ITPStream(Stream):
    def __init__(self, **kwargs):
        super().__init__(**kwargs)
        self.period_in_slot = 0
        self.latency_req_in_slot = 0
        self.latency_requirement = kwargs['latency_requirement']
        self.pkt_num = kwargs['pkt_num']
